package com.zhentao.wemedia.interceptor;


import com.zhentao.model.common.wemedia.WmUser;
import com.zhentao.utils.wemedia.WmThreadLocalUtil;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.Optional;

/**
 * @Date 2024/2/24 10:59
 * @Author xdm
 **/
@Slf4j
public class WmTokenInterceptor implements HandlerInterceptor {
    //这个方法将在请求处理之前进行调用。
    // 注意：如果该方法的返回值为false , 将视为当前请求结束，
    // 不仅自身的拦截器会失效，还会导致其他的拦截器也不再执行
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
       String userId= request.getHeader("userId");
        if(userId!=null){
            //把userId存放到threadLocal中
            WmUser wmUser=new WmUser();
            wmUser.setId(Integer.valueOf(userId));
            WmThreadLocalUtil.setUser(wmUser);
            log.info("WmTokenInterceptor把用户信息放到threadLocal中{}",wmUser);
        }

        return true;
    }

    /*
    只有在 preHandle() 方法返回值为true 时才会执行。
    会在Controller 中的方法调用之后，DispatcherServlet 返回渲染视图之前被调用。
     */
    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
      //清理threadLocal
        WmThreadLocalUtil.clear();
    }

    /*
    只有在 preHandle() 方法返回值为true 时才会执行。在整个请求结束之后，
    DispatcherServlet 渲染了对应的视图之后执行。
     */
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        HandlerInterceptor.super.afterCompletion(request, response, handler, ex);
    }
}